home *** CD-ROM | disk | FTP | other *** search
/ PCGUIA 2010 Software/Programs / PCGuia_programas.iso / Software / Linux / Fedora 12 / Installer / Fedora-12-i686-Live.iso / LiveOS / livecd-iso-to-disk next >
Encoding:
Text File  |  2009-11-09  |  22.5 KB  |  764 lines
  1. #!/bin/bash
  2. # Convert a live CD iso so that it's bootable off of a USB stick
  3. # Copyright 2007  Red Hat, Inc.
  4. # Jeremy Katz <katzj@redhat.com>
  5. #
  6. # overlay/persistence enhancements by Douglas McClendon <dmc@viros.org>
  7. # GPT+MBR hybrid enhancements by Stewart Adam <s.adam@diffingo.com>
  8. # 
  9. # This program is free software; you can redistribute it and/or modify
  10. # it under the terms of the GNU General Public License as published by
  11. # the Free Software Foundation; version 2 of the License.
  12. #
  13. # This program is distributed in the hope that it will be useful,
  14. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16. # GNU Library General Public License for more details.
  17. #
  18. # You should have received a copy of the GNU General Public License
  19. # along with this program; if not, write to the Free Software
  20. # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  21.  
  22.  
  23. export PATH=/sbin:/usr/sbin:$PATH
  24.  
  25. usage() {
  26.     echo "$0 [--format] [--reset-mbr] [--noverify] [--overlay-size-mb <size>] [--home-size-mb <size>] [--unencrypted-home] [--skipcopy] <isopath> <usbstick device>"
  27.     exit 1
  28. }
  29.  
  30. cleanup() {
  31.     sleep 2
  32.     [ -d "$CDMNT" ] && umount $CDMNT && rmdir $CDMNT
  33.     [ -d "$USBMNT" ] && umount $USBMNT && rmdir $USBMNT
  34. }
  35.  
  36. exitclean() {
  37.     echo "Cleaning up to exit..."
  38.     cleanup
  39.     exit 1
  40. }
  41.  
  42. getdisk() {
  43.     DEV=$1
  44.  
  45.     if [[ "$DEV" =~ "/dev/loop*" ]]; then
  46.        device="$DEV"
  47.        return
  48.     fi
  49.  
  50.     p=$(udevadm info -q path -n $DEV)
  51.     if [ -e /sys/$p/device ]; then
  52.     device=$(basename /sys/$p)
  53.     else
  54.     device=$(basename $(readlink -f /sys/$p/../))
  55.     fi
  56.     if [ ! -e /sys/block/$device -o ! -e /dev/$device ]; then
  57.     echo "Error finding block device of $DEV.  Aborting!"
  58.     exitclean
  59.     fi
  60.  
  61.     device="/dev/$device"
  62.     # FIXME: weird dev names could mess this up I guess
  63.     p=/dev/`basename $p`
  64.     partnum=${p##$device}
  65. }
  66.  
  67. resetMBR() {
  68.     if [[ "$DEV" =~ "/dev/loop*" ]]; then
  69.        return
  70.     fi
  71.     getdisk $1
  72.     # if efi, we need to use the hybrid MBR
  73.     if [ -n "$efi" ];then
  74.       if [ -f /usr/lib/syslinux/gptmbr.bin ]; then
  75.         gptmbr='/usr/lib/syslinux/gptmbr.bin'
  76.       elif [ -f /usr/share/syslinux/gptmbr.bin ]; then
  77.         gptmbr='/usr/share/syslinux/gptmbr.bin'
  78.       else
  79.         echo "Could not find gptmbr.bin (syslinux)"
  80.         exitclean
  81.       fi
  82.       # our magic number is LBA-2, offset 16 - (512+512+16)/$bs
  83.       dd if=$device bs=16 skip=65 count=1 | cat $gptmbr - > $device
  84.     else
  85.       if [ -f /usr/lib/syslinux/mbr.bin ]; then
  86.         cat /usr/lib/syslinux/mbr.bin > $device
  87.       elif [ -f /usr/share/syslinux/mbr.bin ]; then
  88.         cat /usr/share/syslinux/mbr.bin > $device
  89.       else
  90.         echo "Could not find mbr.bin (syslinux)"
  91.         exitclean
  92.       fi
  93.     fi
  94. }
  95.  
  96. checkMBR() {
  97.     if [[ "$DEV" =~ "/dev/loop*" ]]; then
  98.        return 0
  99.     fi
  100.     getdisk $1
  101.  
  102.     bs=$(mktemp /tmp/bs.XXXXXX)
  103.     dd if=$device of=$bs bs=512 count=1 2>/dev/null || exit 2
  104.     
  105.     mbrword=$(hexdump -n 2 $bs |head -n 1|awk {'print $2;'})
  106.     rm -f $bs
  107.     if [ "$mbrword" = "0000" ]; then
  108.     echo "MBR appears to be blank."
  109.     echo "Do you want to replace the MBR on this device?"
  110.     echo "Press Enter to continue or ctrl-c to abort"
  111.     read
  112.     resetMBR $1
  113.     fi
  114.  
  115.     return 0
  116. }
  117.  
  118. checkPartActive() {
  119.     dev=$1
  120.     getdisk $dev
  121.     
  122.     # if we're installing to whole-disk and not a partition, then we 
  123.     # don't need to worry about being active
  124.     if [ "$dev" = "$device" ]; then
  125.     return
  126.     fi
  127.     if [[ "$dev" =~ "/dev/loop*" ]]; then
  128.         return
  129.     fi
  130.  
  131.     if [ "$(/sbin/fdisk -l $device 2>/dev/null |grep $dev |awk {'print $2;'})" != "*" ]; then
  132.     echo "Partition isn't marked bootable!"
  133.     echo "You can mark the partition as bootable with "
  134.         echo "    # /sbin/parted $device"
  135.     echo "    (parted) toggle N boot"
  136.     echo "    (parted) quit"
  137.     exitclean
  138.     fi
  139. }
  140.  
  141. createGPTLayout() {
  142.     dev=$1
  143.     getdisk $dev
  144.  
  145.     echo "WARNING: THIS WILL DESTROY ANY DATA ON $device!!!"
  146.     echo "Press Enter to continue or ctrl-c to abort"
  147.     read
  148.     umount ${device}? &> /dev/null
  149.     /sbin/parted --script $device mklabel gpt
  150.     partinfo=$(/sbin/parted --script -m $device "unit b print" |grep ^$device:)
  151.     size=$(echo $partinfo |cut -d : -f 2 |sed -e 's/B$//')
  152.     /sbin/parted --script $device unit b mkpart '"EFI System Partition"' fat32 17408 $(($size - 17408)) set 1 boot on
  153.     USBDEV=${device}1
  154.     # Sometimes automount can be _really_ annoying.
  155.     echo "Waiting for devices to settle..."
  156.     /sbin/udevadm settle
  157.     sleep 5
  158.     umount $USBDEV &> /dev/null
  159.     /sbin/mkdosfs -n LIVE $USBDEV
  160.     USBLABEL="UUID=$(/sbin/blkid -s UUID -o value $USBDEV)"
  161. }
  162.  
  163. createMSDOSLayout() {
  164.     dev=$1
  165.     getdisk $dev
  166.  
  167.     echo "WARNING: THIS WILL DESTROY ANY DATA ON $device!!!"
  168.     echo "Press Enter to continue or ctrl-c to abort"
  169.     read
  170.     umount ${device}? &> /dev/null
  171.     /sbin/parted --script $device mklabel msdos
  172.     partinfo=$(/sbin/parted --script -m $device "unit b print" |grep ^$device:)
  173.     size=$(echo $partinfo |cut -d : -f 2 |sed -e 's/B$//')
  174.     /sbin/parted --script $device unit b mkpart primary fat32 17408 $(($size - 17408)) set 1 boot on
  175.     USBDEV=${device}1
  176.     # Sometimes automount can be _really_ annoying.
  177.     echo "Waiting for devices to settle..."
  178.     /sbin/udevadm settle
  179.     sleep 5
  180.     umount $USBDEV &> /dev/null
  181.     /sbin/mkdosfs -n LIVE $USBDEV
  182.     USBLABEL="UUID=$(/sbin/blkid -s UUID -o value $USBDEV)"
  183. }
  184.  
  185. checkGPT() {
  186.     dev=$1
  187.     getdisk $dev
  188.  
  189.     if [ "$(/sbin/fdisk -l $device 2>/dev/null |grep -c GPT)" -eq "0" ]; then
  190.        echo "EFI boot requires a GPT partition table."
  191.        echo "This can be done manually or you can run with --format"
  192.        exitclean
  193.     fi
  194.  
  195.     partinfo=$(/sbin/parted --script -m $device "print" |grep ^$partnum:)
  196.     volname=$(echo $partinfo |cut -d : -f 6)
  197.     flags=$(echo $partinfo |cut -d : -f 7)
  198.     if [ "$volname" != "EFI System Partition" ]; then
  199.     echo "Partition name must be 'EFI System Partition'"
  200.     echo "This can be set in parted or you can run with --reset-mbr"
  201.     exitclean
  202.     fi
  203.     if [ "$(echo $flags |grep -c boot)" = "0" ]; then
  204.     echo "Partition isn't marked bootable!"
  205.     echo "You can mark the partition as bootable with "
  206.         echo "    # /sbin/parted $device"
  207.     echo "    (parted) toggle N boot"
  208.     echo "    (parted) quit"
  209.     exitclean
  210.     fi
  211. }
  212.  
  213. checkFilesystem() {
  214.     dev=$1
  215.  
  216.     USBFS=$(/sbin/blkid -s TYPE -o value $dev)
  217.     if [ "$USBFS" != "vfat" -a "$USBFS" != "msdos" -a "$USBFS" != "ext2" -a "$USBFS" != "ext3" ]; then
  218.     echo "USB filesystem must be vfat or ext[23]"
  219.     exitclean
  220.     fi
  221.  
  222.     USBLABEL=$(/sbin/blkid -s UUID -o value $dev)
  223.     if [ -n "$USBLABEL" ]; then 
  224.     USBLABEL="UUID=$USBLABEL" ; 
  225.     else
  226.     USBLABEL=$(/sbin/blkid -s LABEL -o value $dev)
  227.     if [ -n "$USBLABEL" ]; then 
  228.         USBLABEL="LABEL=$USBLABEL" 
  229.     else
  230.         echo "Need to have a filesystem label or UUID for your USB device"
  231.         if [ "$USBFS" = "vfat" -o "$USBFS" = "msdos" ]; then
  232.         echo "Label can be set with /sbin/dosfslabel"
  233.         elif [ "$USBFS" = "ext2" -o "$USBFS" = "ext3" ]; then
  234.         echo "Label can be set with /sbin/e2label"
  235.         fi
  236.         exitclean
  237.     fi
  238.     fi
  239.  
  240.     if [ "$USBFS" = "vfat" -o "$USBFS" = "msdos" ]; then
  241.     mountopts="-o shortname=winnt,umask=0077"
  242.     fi
  243. }
  244.  
  245. checkSyslinuxVersion() {
  246.     if [ ! -x /usr/bin/syslinux ]; then
  247.     echo "You need to have syslinux installed to run this script"
  248.     exit 1
  249.     fi
  250.     if ! syslinux 2>&1 | grep -qe -d; then
  251.     SYSLINUXPATH=""
  252.     elif [ -n "$multi" ]; then
  253.     SYSLINUXPATH="$LIVEOS/syslinux"
  254.     else
  255.     SYSLINUXPATH="syslinux"
  256.     fi
  257. }
  258.  
  259. checkMounted() {
  260.     dev=$1
  261.     if grep -q "^$dev " /proc/mounts ; then
  262.       echo "$dev is mounted, please unmount for safety"
  263.       exitclean
  264.     fi
  265.     if grep -q "^$dev " /proc/swaps; then
  266.       echo "$dev is in use as a swap device, please disable swap"
  267.       exitclean
  268.     fi
  269. }
  270.  
  271. checkint() {
  272.     if ! test $1 -gt 0 2>/dev/null ; then
  273.     usage
  274.     fi
  275. }
  276.  
  277. if [ $(id -u) != 0 ]; then 
  278.     echo "You need to be root to run this script"
  279.     exit 1
  280. fi
  281.  
  282. detectisotype() {
  283.     if [ -e $CDMNT/LiveOS/squashfs.img ]; then
  284.         isotype=live
  285.         return
  286.     fi
  287.     if [ -e $CDMNT/images/install.img ]; then
  288.         isotype=installer
  289.         return
  290.     fi
  291.     echo "ERROR: $ISO does not appear to be a Live image or DVD installer."
  292.     exitclean
  293. }
  294.  
  295. cryptedhome=1
  296. keephome=1
  297. homesizemb=0
  298. swapsizemb=0
  299. overlaysizemb=0
  300. isotype=
  301. LIVEOS=LiveOS
  302.  
  303. HOMEFILE="home.img"
  304. while [ $# -gt 2 ]; do
  305.     case $1 in
  306.     --overlay-size-mb)
  307.         checkint $2
  308.         overlaysizemb=$2
  309.         shift
  310.         ;;
  311.     --home-size-mb)
  312.         checkint $2
  313.             homesizemb=$2
  314.             shift
  315.         ;;
  316.     --swap-size-mb)
  317.         checkint $2
  318.         swapsizemb=$2
  319.         shift
  320.         ;;
  321.         --crypted-home)
  322.             cryptedhome=1
  323.         ;;
  324.         --unencrypted-home)
  325.             cryptedhome=""
  326.             ;;
  327.         --delete-home)
  328.             keephome=""
  329.             ;;
  330.     --noverify)
  331.         noverify=1
  332.         ;;
  333.     --reset-mbr|--resetmbr)
  334.         resetmbr=1
  335.         ;;
  336.     --efi|--mactel)
  337.         efi=1
  338.         ;;
  339.     --format)
  340.         format=1
  341.         ;;
  342.     --skipcopy)
  343.         skipcopy=1
  344.         ;;
  345.     --xo)
  346.         xo=1
  347.         skipcompress=1
  348.         ;;
  349.     --xo-no-home)
  350.         xonohome=1
  351.         ;;
  352.     --compress)
  353.         skipcompress=""
  354.         ;;
  355.     --skipcompress)
  356.         skipcompress=1
  357.         ;;
  358.         --extra-kernel-args)
  359.             kernelargs=$2
  360.             shift
  361.             ;;
  362.         --force)
  363.             force=1
  364.             ;;
  365.     --livedir)
  366.         LIVEOS=$2
  367.         shift
  368.         ;;
  369.     --multi)
  370.         multi=1
  371.         ;;
  372.     *)
  373.         echo "invalid arg -- $1"
  374.         usage
  375.         ;;
  376.     esac
  377.     shift
  378. done
  379.  
  380. ISO=$(readlink -f "$1")
  381. USBDEV=$(readlink -f "$2")
  382.  
  383. if [ -z "$ISO" ]; then
  384.     usage
  385. fi
  386.  
  387. if [ ! -b "$ISO" -a ! -f "$ISO" ]; then
  388.     usage
  389. fi
  390.  
  391. # FIXME: If --format is given, we shouldn't care and just use /dev/foo1
  392. if [ -z "$USBDEV" -o ! -b "$USBDEV" ]; then
  393.     usage
  394. fi
  395.  
  396. if [ -z "$noverify" ]; then
  397.     # verify the image
  398.     echo "Verifying image..."
  399.     checkisomd5 --verbose "$ISO"
  400.     if [ $? -ne 0 ]; then
  401.     echo "Are you SURE you want to continue?"
  402.     echo "Press Enter to continue or ctrl-c to abort"
  403.     read
  404.     fi
  405. fi
  406.  
  407. # do some basic sanity checks.  
  408. checkMounted $USBDEV
  409. if [ -n "$format" ];then
  410.   # checks for a valid filesystem
  411.   if [ -n "$efi" ];then
  412.     createGPTLayout $USBDEV
  413.   else
  414.     createMSDOSLayout $USBDEV
  415.   fi
  416. fi
  417. checkFilesystem $USBDEV
  418. if [ -n "$efi" ]; then
  419.   checkGPT $USBDEV
  420. fi
  421. checkSyslinuxVersion
  422. # Because we can't set boot flag for EFI Protective on msdos partition tables
  423. [ -z "$efi" ] && checkPartActive $USBDEV
  424. [ -n "$resetmbr" ] && resetMBR $USBDEV
  425. checkMBR $USBDEV
  426.  
  427.  
  428. if [ "$overlaysizemb" -gt 0 -a "$USBFS" = "vfat" ]; then
  429.   if [ "$overlaysizemb" -gt 2047 ]; then
  430.     echo "Can't have an overlay of 2048MB or greater on VFAT"
  431.     exitclean
  432.   fi
  433. fi
  434.  
  435. if [ "$homesizemb" -gt 0 -a "$USBFS" = "vfat" ]; then
  436.   if [ "$homesizemb" -gt 2047 ]; then
  437.     echo "Can't have a home overlay greater than 2048MB on VFAT"
  438.     exitclean
  439.   fi
  440. fi
  441.  
  442. if [ "$swapsizemb" -gt 0 -a "$USBFS" = "vfat" ]; then
  443.   if [ "$swapsizemb" -gt 2047 ]; then
  444.     echo "Can't have a swap file greater than 2048MB on VFAT"
  445.     exitclean
  446.   fi
  447. fi
  448.  
  449. # FIXME: would be better if we had better mountpoints
  450. CDMNT=$(mktemp -d /media/cdtmp.XXXXXX)
  451. mount -o loop,ro "$ISO" $CDMNT || exitclean
  452. USBMNT=$(mktemp -d /media/usbdev.XXXXXX)
  453. mount $mountopts $USBDEV $USBMNT || exitclean
  454.  
  455. trap exitclean SIGINT SIGTERM
  456.  
  457. detectisotype
  458.  
  459. if [ -f "$USBMNT/$LIVEOS/$HOMEFILE" -a -n "$keephome" -a "$homesizemb" -gt 0 ]; then
  460.   echo "ERROR: Requested keeping existing /home and specified a size for /home"
  461.   echo "Please either don't specify a size or specify --delete-home"
  462.   exitclean
  463. fi
  464.  
  465. if [ -n "$efi" -a ! -d $CDMNT/EFI/boot ]; then
  466.   echo "ERROR: This live image does not support EFI booting"
  467.   exitclean
  468. fi
  469.  
  470. # let's try to make sure there's enough room on the stick
  471. if [ -d $CDMNT/LiveOS ]; then
  472.   check=$CDMNT/LiveOS
  473. else
  474.   check=$CDMNT
  475. fi
  476. if [ -d $USBMNT/$LIVEOS ]; then
  477.   tbd=$(du -s -B 1M $USBMNT/$LIVEOS | awk {'print $1;'})
  478.   [ -f $USBMNT/$LIVEOS/$HOMEFILE ] && homesz=$(du -s -B 1M $USBMNT/$LIVEOS/$HOMEFILE | awk {'print $1;'})
  479.   [ -n "$homesz" -a -n "$keephome" ] && tbd=$(($tbd - $homesz))
  480. else
  481.   tbd=0
  482. fi
  483. livesize=$(du -s -B 1M $check | awk {'print $1;'})
  484. if [ -n "$skipcompress" ]; then
  485.     if [ -e $CDMNT/LiveOS/squashfs.img ]; then
  486.     if mount -o loop $CDMNT/LiveOS/squashfs.img $CDMNT; then
  487.         livesize=$(du -s -B 1M $CDMNT/LiveOS/ext3fs.img | awk {'print $1;'})
  488.         umount $CDMNT
  489.     else
  490.         echo "WARNING: --skipcompress or --xo was specified but the currently"
  491.         echo "running kernel can not mount the squashfs from the ISO file to extract"
  492.         echo "it. The compressed squashfs will be copied to the USB stick."
  493.         skipcompress=""
  494.     fi
  495.     fi
  496. fi
  497. free=$(df  -B1M $USBDEV  |tail -n 1 |awk {'print $4;'})
  498.  
  499. if [ "$isotype" = "live" ]; then
  500. tba=$(($overlaysizemb + $homesizemb + $livesize + $swapsizemb))
  501. if [ $tba -gt $(($free + $tbd)) ]; then
  502.   echo "Unable to fit live image + overlay on available space on USB stick"
  503.   echo "+ Size of live image:  $livesize"
  504.   [ "$overlaysizemb" -gt 0 ] && echo "+ Overlay size:  $overlaysizemb"
  505.   [ "$homesizemb" -gt 0 ] && echo "+ Home overlay size:  $homesizemb"
  506.   [ "$swapsizemb" -gt 0 ] && echo "+ Swap overlay size:  $swapsizemb"
  507.   echo "---------------------------"
  508.   echo "= Requested:  $tba"
  509.   echo "- Available:  $(($free + $tbd))"
  510.   echo "---------------------------"
  511.   echo "= To fit, free or decrease requested size total by:  $(($tba - $free - $tbd))"
  512.   exitclean
  513. fi
  514. fi
  515.  
  516. # Verify available space for DVD installer 
  517. if [ "$isotype" = "installer" ]; then
  518.   isosize=$(du -s -B 1M $ISO | awk {'print $1;'})
  519.   installimgsize=$(du -s -B 1M $CDMNT/images/install.img | awk {'print $1;'})
  520.   tbd=0
  521.   if [ -e $USBMNT/images/install.img ]; then
  522.     tbd=$(du -s -B 1M $USBMNT/images/install.img | awk {'print $1;'})
  523.   fi
  524.   if [ -e $USBMNT/$(basename $ISO) ]; then
  525.     tbd=$(($tbd + $(du -s -B 1M $USBMNT/$(basename $ISO) | awk {'print $1;'})))
  526.   fi
  527.   echo "Size of DVD image: $isosize"
  528.   echo "Size of install.img: $installimgsize"
  529.   echo "Available space: $(($free + $tbd))"
  530.   if [ $(($isosize + $installimgsize)) -gt $(($free + $tbd)) ]; then
  531.     echo "ERROR: Unable to fit DVD image + install.img on available space on USB stick"
  532.     exitclean
  533.   fi
  534. fi
  535.  
  536. if [ -z "$skipcopy" ] && [ "$isotype" = "live" ]; then
  537.   if [ -d $USBMNT/$LIVEOS -a -z "$force" ]; then
  538.       echo "Already set up as live image."  
  539.       if [ -z "$keephome" -a -e $USBMNT/$LIVEOS/$HOMEFILE ]; then
  540.         echo "WARNING: Persistent /home will be deleted!!!"
  541.         echo "Press Enter to continue or ctrl-c to abort"
  542.         read
  543.       else
  544.         echo "Deleting old OS in fifteen seconds..."
  545.         sleep 15
  546.  
  547.         [ -e "$USBMNT/$LIVEOS/$HOMEFILE" -a -n "$keephome" ] && mv $USBMNT/$LIVEOS/$HOMEFILE $USBMNT/$HOMEFILE
  548.       fi
  549.  
  550.       rm -rf $USBMNT/$LIVEOS
  551.   fi
  552. fi
  553.  
  554. # Bootloader is always reconfigured, so keep these out of the if skipcopy stuff.
  555. [ ! -d $USBMNT/$SYSLINUXPATH ] && mkdir -p $USBMNT/$SYSLINUXPATH
  556. [ -n "$efi" -a ! -d $USBMNT/EFI/boot ] && mkdir -p $USBMNT/EFI/boot
  557.  
  558. # Live image copy
  559. if [ -z "$skipcopy" ] && [ "$isotype" = "live" ]; then
  560.   echo "Copying live image to USB stick"
  561.   [ ! -d $USBMNT/$LIVEOS ] && mkdir $USBMNT/$LIVEOS
  562.   [ -n "$keephome" -a -f "$USBMNT/$HOMEFILE" ] && mv $USBMNT/$HOMEFILE $USBMNT/$LIVEOS/$HOMEFILE
  563.   if [ -n "$skipcompress" -a -f $CDMNT/LiveOS/squashfs.img ]; then
  564.       mount -o loop $CDMNT/LiveOS/squashfs.img $CDMNT || exitclean
  565.       cp $CDMNT/LiveOS/ext3fs.img $USBMNT/$LIVEOS/ext3fs.img || (umount $CDMNT ; exitclean)
  566.       umount $CDMNT
  567.   elif [ -f $CDMNT/LiveOS/squashfs.img ]; then
  568.       cp $CDMNT/LiveOS/squashfs.img $USBMNT/$LIVEOS/squashfs.img || exitclean
  569.   elif [ -f $CDMNT/LiveOS/ext3fs.img ]; then
  570.       cp $CDMNT/LiveOS/ext3fs.img $USBMNT/$LIVEOS/ext3fs.img || exitclean
  571.   fi
  572.   if [ -f $CDMNT/LiveOS/osmin.img ]; then
  573.       cp $CDMNT/LiveOS/osmin.img $USBMNT/$LIVEOS/osmin.img || exitclean
  574.   fi
  575. fi
  576.  
  577. # DVD installer copy
  578. if [ "$isotype" = "installer" ] && [ -z "$skipcopy" ]; then
  579.       echo "Copying DVD image to USB stick"
  580.       mkdir -p $USBMNT/images/
  581.       cp $CDMNT/images/install.img $USBMNT/images/install.img || exitclean
  582.       cp $ISO $USBMNT/
  583. fi
  584.  
  585. cp $CDMNT/isolinux/* $USBMNT/$SYSLINUXPATH
  586. BOOTCONFIG=$USBMNT/$SYSLINUXPATH/isolinux.cfg
  587. # Set this to nothing so sed doesn't care
  588. BOOTCONFIG_EFI=
  589. if [ -n "$efi" ];then
  590.   cp $CDMNT/EFI/boot/* $USBMNT/EFI/boot
  591.  
  592.   # this is a little ugly, but it gets the "interesting" named config file
  593.   BOOTCONFIG_EFI=$USBMNT/EFI/boot/boot?*.conf
  594.   rm -f $USBMNT/EFI/boot/grub.conf
  595. fi
  596.  
  597. echo "Updating boot config file"
  598. # adjust label and fstype
  599. sed -i -e "s/CDLABEL=[^ ]*/$USBLABEL/" -e "s/rootfstype=[^ ]*/rootfstype=$USBFS/" $BOOTCONFIG  $BOOTCONFIG_EFI
  600. if [ -n "$kernelargs" ]; then sed -i -e "s/liveimg/liveimg ${kernelargs}/" $BOOTCONFIG $BOOTCONFIG_EFI ; fi
  601. if [ "$LIVEOS" != "LiveOS" ]; then sed -i -e "s;liveimg;liveimg live_dir=$LIVEOS;" $BOOTCONFIG $BOOTCONFIG_EFI ; fi
  602.  
  603. # DVD Installer
  604. if [ "$isotype" = "installer" ]; then
  605.   sed -i -e "s;initrd=initrd.img;initrd=initrd.img repo=hd:$USBLABEL:/;g" $BOOTCONFIG $BOOTCONFIG_EFI
  606.   sed -i -e "s;stage2=\S*;;g" $BOOTCONFIG $BOOTCONFIG_EFI
  607. fi
  608.  
  609. if [ "$overlaysizemb" -gt 0 ]; then
  610.     echo "Initializing persistent overlay file"
  611.     OVERFILE="overlay-$( /sbin/blkid -s LABEL -o value $USBDEV )-$( /sbin/blkid -s UUID -o value $USBDEV )"
  612.     if [ "$USBFS" = "vfat" ]; then
  613.     # vfat can't handle sparse files
  614.     dd if=/dev/zero of=$USBMNT/$LIVEOS/$OVERFILE count=$overlaysizemb bs=1M
  615.     else
  616.     dd if=/dev/null of=$USBMNT/$LIVEOS/$OVERFILE count=1 bs=1M seek=$overlaysizemb
  617.     fi
  618.     sed -i -e "s/liveimg/liveimg overlay=${USBLABEL}/" $BOOTCONFIG $BOOTCONFIG_EFI
  619.     sed -i -e "s/\ ro\ /\ rw\ /" $BOOTCONFIG  $BOOTCONFIG_EFI
  620. fi
  621.  
  622. if [ "$swapsizemb" -gt 0 ]; then
  623.     echo "Initializing swap file"
  624.     dd if=/dev/zero of=$USBMNT/$LIVEOS/swap.img count=$swapsizemb bs=1M
  625.     mkswap -f $USBMNT/$LIVEOS/swap.img
  626. fi
  627.  
  628. if [ "$homesizemb" -gt 0 ]; then
  629.     echo "Initializing persistent /home"
  630.     homesource=/dev/zero
  631.     [ -n "$cryptedhome" ] && homesource=/dev/urandom
  632.     if [ "$USBFS" = "vfat" ]; then
  633.     # vfat can't handle sparse files
  634.     dd if=${homesource} of=$USBMNT/$LIVEOS/$HOMEFILE count=$homesizemb bs=1M
  635.     else
  636.     dd if=/dev/null of=$USBMNT/$LIVEOS/$HOMEFILE count=1 bs=1M seek=$homesizemb
  637.     fi
  638.     if [ -n "$cryptedhome" ]; then
  639.     loop=$(losetup -f)
  640.     losetup $loop $USBMNT/$LIVEOS/$HOMEFILE
  641.     setupworked=1
  642.     until [ ${setupworked} == 0 ]; do
  643.             echo "Encrypting persistent /home"
  644.             cryptsetup luksFormat -y -q $loop
  645.         setupworked=$?
  646.     done
  647.     setupworked=1
  648.     until [ ${setupworked} == 0 ]; do
  649.             echo "Please enter the password again to unlock the device"
  650.             cryptsetup luksOpen $loop EncHomeFoo
  651.         setupworked=$?
  652.     done
  653.         mke2fs -j /dev/mapper/EncHomeFoo
  654.     tune2fs -c0 -i0 -ouser_xattr,acl /dev/mapper/EncHomeFoo
  655.     sleep 2
  656.         cryptsetup luksClose EncHomeFoo
  657.         losetup -d $loop
  658.     else
  659.         echo "Formatting unencrypted /home"
  660.     mke2fs -F -j $USBMNT/$LIVEOS/$HOMEFILE
  661.     tune2fs -c0 -i0 -ouser_xattr,acl $USBMNT/$LIVEOS/$HOMEFILE
  662.     fi
  663. fi
  664.  
  665. # create the forth files for booting on the XO if requested
  666. # we'd do this unconditionally, but you have to have a kernel that will
  667. # boot on the XO anyway.
  668. if [ -n "$xo" ]; then
  669.     echo "Setting up /boot/olpc.fth file"
  670.     args=$(egrep "^[ ]*append" $USBMNT/$SYSLINUXPATH/isolinux.cfg |head -n1 |sed -e 's/.*initrd=[^ ]*//')
  671.     if [ -z "$xonohome" -a ! -f $USBMNT/$LIVEOS/$HOMEFILE ]; then
  672.     args="$args persistenthome=mtd0"
  673.     fi
  674.     args="$args reset_overlay"
  675.     xosyspath=$(echo $SYSLINUXPATH | sed -e 's;/;\\;')
  676.     if [ ! -d $USBMNT/boot ]; then mkdir -p $USBMNT/boot ; fi
  677.     cat > $USBMNT/boot/olpc.fth <<EOF
  678. \ Boot script for USB boot
  679. hex  rom-pa fffc7 + 4 \$number drop  h# 2e19 < [if]
  680.   patch 2drop erase claim-params
  681.   : high-ramdisk  ( -- )
  682.      cv-load-ramdisk
  683.      h# 22c +lp l@ 1+   memory-limit  umin  /ramdisk - ffff.f000 and ( new-ramdisk-adr )
  684.      ramdisk-adr over  /ramdisk move                    ( new-ramdisk-adr )
  685.      to ramdisk-adr
  686.   ;
  687.   ' high-ramdisk to load-ramdisk
  688. [then]
  689.  
  690. : set-bootpath-dev  ( -- )
  691.    " /chosen" find-package  if                       ( phandle )
  692.       " bootpath" rot  get-package-property  0=  if  ( propval$ )
  693.          get-encoded-string                          ( bootpath$ )
  694.          [char] \ left-parse-string  2nip            ( dn$ )
  695.          dn-buf place                                ( )
  696.       then
  697.    then
  698.  
  699.    " /sd"  dn-buf  count  sindex  0>=   if
  700.           " sd:"
  701.    else
  702.           " u:"
  703.    then
  704.    " BOOTPATHDEV" \$set-macro
  705. ;
  706.  
  707. set-bootpath-dev
  708. " $args" to boot-file
  709. " \${BOOTPATHDEV}$xosyspath\initrd0.img" expand$ to ramdisk
  710. " \${BOOTPATHDEV}$xosyspath\vmlinuz0" expand$ to boot-device
  711. unfreeze
  712. boot
  713. EOF
  714.  
  715. fi
  716.  
  717. if [ -z "$multi" ]; then
  718.   echo "Installing boot loader"
  719.   if [ -n "$efi" ]; then
  720.     # replace the ia32 hack
  721.     if [ -f "$USBMNT/EFI/boot/boot.conf" ]; then cp -f $USBMNT/EFI/boot/bootia32.conf $USBMNT/EFI/boot/boot.conf ; fi
  722.   fi
  723.  
  724.   # this is a bit of a kludge, but syslinux doesn't guarantee the API for its com32 modules :/
  725.   if [ -f $USBMNT/$SYSLINUXPATH/vesamenu.c32 -a -f /usr/share/syslinux/vesamenu.c32 ]; then
  726.     cp /usr/share/syslinux/vesamenu.c32 $USBMNT/$SYSLINUXPATH/vesamenu.c32
  727.   elif [ -f $USBMNT/$SYSLINUXPATH/vesamenu.c32 -a -f /usr/lib/syslinux/vesamenu.c32 ]; then
  728.     cp /usr/lib/syslinux/vesamenu.c32 $USBMNT/$SYSLINUXPATH/vesamenu.c32
  729.   elif [ -f $USBMNT/$SYSLINUXPATH/menu.c32 -a -f /usr/share/syslinux/menu.c32 ]; then
  730.     cp /usr/share/syslinux/menu.c32 $USBMNT/$SYSLINUXPATH/menu.c32
  731.   elif [ -f $USBMNT/$SYSLINUXPATH/menu.c32 -a -f /usr/lib/syslinux/menu.c32 ]; then
  732.     cp /usr/lib/syslinux/menu.c32 $USBMNT/$SYSLINUXPATH/menu.c32
  733.   fi
  734.  
  735.   if [ "$USBFS" = "vfat" -o "$USBFS" = "msdos" ]; then
  736.     # syslinux expects the config to be named syslinux.cfg 
  737.     # and has to run with the file system unmounted
  738.     mv $USBMNT/$SYSLINUXPATH/isolinux.cfg $USBMNT/$SYSLINUXPATH/syslinux.cfg
  739.     # deal with mtools complaining about ldlinux.sys
  740.     if [ -f $USBMNT/$SYSLINUXPATH/ldlinux.sys ] ; then rm -f $USBMNT/$SYSLINUXPATH/ldlinux.sys ; fi
  741.     cleanup
  742.     if [ -n "$SYSLINUXPATH" ]; then
  743.       syslinux -d $SYSLINUXPATH $USBDEV
  744.     else
  745.       syslinux $USBDEV
  746.     fi
  747.   elif [ "$USBFS" = "ext2" -o "$USBFS" = "ext3" ]; then
  748.     # extlinux expects the config to be named extlinux.conf
  749.     # and has to be run with the file system mounted
  750.     mv $USBMNT/$SYSLINUXPATH/isolinux.cfg $USBMNT/$SYSLINUXPATH/extlinux.conf
  751.     extlinux -i $USBMNT/$SYSLINUXPATH
  752.     chattr -i $USBMNT/$SYSLINUXPATH/extlinux.sys
  753.     cleanup
  754.   fi
  755. else
  756.   # we need to do some more config file tweaks for multi-image mode
  757.   sed -i -e "s;kernel vm;kernel /$LIVEOS/syslinux/vm;" $USBMNT/$SYSLINUXPATH/isolinux.cfg
  758.   sed -i -e "s;initrd=i;initrd=/$LIVEOS/syslinux/i;" $USBMNT/$SYSLINUXPATH/isolinux.cfg
  759.   mv $USBMNT/$SYSLINUXPATH/isolinux.cfg $USBMNT/$SYSLINUXPATH/syslinux.cfg
  760.   cleanup
  761. fi
  762.  
  763. echo "USB stick set up as live image!"
  764.